import unittest
import app
import time

from script.test_ihrm_emp_manage_params import TestEmpManageParams
from script.test_ihrm_login_params import TestIHRMLoginParams
from lib.HTMLTestRunner2 import HTMLTestRunner

# 1. 创建测试套件类对象
suite = unittest.TestSuite()

# 2. 使用测试套件类的对象，调用 addTest() 方法，添加测试用例 --- 逐类添加所有方法
suite.addTest(unittest.makeSuite(TestIHRMLoginParams))
suite.addTest(unittest.makeSuite(TestEmpManageParams))

# 3. 组织测试报告的文件名   ihrm.html
# report_name = app.BASE_PATH + '/report/ihrm2.html'
report_name = app.BASE_PATH + '/report/ihrm{}.html'.format(time.strftime("%Y-%m-%d %H-%M-%S"))

# 4. 以 wb 格式打开文件测试报告文件，
with open(report_name, 'wb') as f:
    # 5. 借助 HTMLTestRunner 模块，创建runner 类对象
    # runner = HTMLTestRunner(f)
    runner = HTMLTestRunner(stream=f, title='iHRM管理系统的测试报告', description='顺义32期全体QA')
    # 6. 使用 runner 的run方法，调用 测试套件，生成测试报告！
    runner.run(suite)